sudo apt install cmake git mpich python3-pip neofetch curl 
pip3 install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio==0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
pip3 install d2l tqdm
sudo apt install nvidia-cuda-toolkit
